home *** CD-ROM | disk | FTP | other *** search
/ Clickx 115 / Clickx 115.iso / software / tools / windows / tails-i386-0.16.iso / live / filesystem.squashfs / usr / share / pyshared / whisperBack / mail.py < prev    next >
Encoding:
Python Source  |  2012-12-13  |  5.6 KB  |  179 lines
  1. #!/usr/bin/env python
  2. # -*- coding: UTF-8 -*-
  3.  
  4. ########################################################################
  5. # WhisperBack - Send feedback in an encrypted mail
  6. # Copyright (C) 2009-2012 Tails developers <amnesia.org>
  7. #
  8. # This file is part of WhisperBack
  9. #
  10. # WhisperBack is  free software; you can redistribute  it and/or modify
  11. # it under the  terms of the GNU General Public  License as published by
  12. # the Free Software Foundation; either  version 3 of the License, or (at
  13. # your option) any later version.
  14. # 
  15. # This program  is distributed in the  hope that it will  be useful, but
  16. # WITHOUT   ANY  WARRANTY;   without  even   the  implied   warranty  of
  17. # MERCHANTABILITY  or FITNESS  FOR A  PARTICULAR PURPOSE.   See  the GNU
  18. # General Public License for more details.
  19. #
  20. # You should have received a copy of the GNU General Public License
  21. # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22. ########################################################################
  23.  
  24. """WhisperBack mailing library
  25.  
  26. """
  27.  
  28. import smtplib
  29. import socket
  30. from email.mime.text import MIMEText
  31. import gnutls.errors
  32. import time
  33.  
  34. def create_message (from_address, to_address, subject, message):
  35.     """Create a plaintext mail
  36.  
  37.     This is from an example from doc.python.org
  38.  
  39.     @param from_address The sender's address
  40.     @param to_address The recipient address
  41.     @param subject The topic
  42.     @param message The content of the text message
  43.  
  44.     @return the message to send
  45.     """
  46.  
  47.     msg = MIMEText(message)
  48.     msg['Subject'] = subject
  49.     msg['From'] = from_address
  50.     msg['To'] = to_address
  51.  
  52.     return msg.as_string()
  53.  
  54. #pylint: disable=R0913
  55. def send_message_tls (from_address, to_address, message, host="localhost",
  56.                     port=25, tls_cafile=None):
  57.     """Sends a mail
  58.  
  59.     This is from an example from doc.python.org
  60.  
  61.     @param from_address The sender's address
  62.     @param to_address The recipient address
  63.     @param message The content of the mail
  64.     @param host The host of the smtp server to connect to
  65.     @param port The port of the smtp server to connect to
  66.     @param tls_ca Certificate authority file passed to the socket moduleΓÇÖs ssl() function.
  67.     """
  68.     # Send the message via our own SMTP server, but don't include the
  69.     # envelope header.
  70.     # We set a long timeout because Tor is slow
  71.     smtp = smtplib.SMTP(timeout=120)
  72.     smtp.connect(host, port)
  73.     (resp, reply) = smtp.starttls(cafile = tls_cafile)
  74.     # Default python let you continue in cleartext if starttls
  75.     # fails, while you expect to have an encrypted connexion
  76.     if resp != 220:
  77.         raise TLSError("%s answered %i, %s when trying to start TLS"
  78.                        % (host, resp, reply))
  79.     else:
  80.         smtp.sendmail(from_address, [to_address], message)
  81.         smtp.quit()
  82.  
  83. class TLSError(Exception):
  84.     """Exception raised if problem happens in STARTTLS step"""
  85.     pass
  86.  
  87. # This is a monkey patch to make the starttls function of libsmtp use
  88. # starttls, as the buildin doesn't really check certificates and doesn't
  89. # have a timeout parameter
  90. #pylint: disable=W0613
  91. def starttls(self, keyfile = None, certfile = None, cafile=None):
  92.     """Puts the connection to the SMTP server into TLS mode.
  93.  
  94.     If the server supports TLS, this will encrypt the rest of the SMTP
  95.     session.
  96.  
  97.     """
  98.     (resp, reply) = self.docmd("STARTTLS")
  99.     if resp == 220:
  100.  
  101.         from gnutls.crypto import X509Certificate
  102.         from gnutls.connection import ClientSession, X509Credentials
  103.         # Don't worry, this is a monkey patch
  104.         #pylint: disable=W0621
  105.         import struct
  106.  
  107.         tv = struct.pack('ii', int(6), int(0))
  108.         self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_RCVTIMEO, tv)
  109.  
  110.         if cafile:
  111.             ca = X509Certificate(open(cafile).read())
  112.         # XXX: use CRL
  113.         #crl = X509CRL(open(certs_path + '/crl.pem').read())
  114.         #cred = X509Credentials(trusted=[ca], crl_list=[crl])
  115.         cred = X509Credentials(trusted=[ca])
  116.         session = ClientSession(self.sock, cred)
  117.  
  118.         while True:
  119.             try:
  120.                 session.handshake()
  121.                 session.verify_peer()
  122.                 break
  123.             except gnutls.errors.OperationWouldBlock:
  124.                 time.sleep(0.1)
  125.  
  126.         def tls_quit():
  127.             """Terminate the SMTP session."""
  128.             self.docmd("quit")
  129.             while True:
  130.                 try:
  131.                     self.sock.bye()
  132.                     break
  133.                 except gnutls.errors.OperationWouldBlock:
  134.                     time.sleep(0.1)
  135.             self.close()
  136.  
  137.         self.quit = tls_quit
  138.  
  139.         self.sock = session
  140.         self.file = SSLFakeFile(session)
  141.  
  142.         # RFC 3207:
  143.         # The client MUST discard any knowledge obtained from
  144.         # the server, such as the list of SMTP service extensions,
  145.         # which was not obtained from the TLS negotiation itself.
  146.         self.helo_resp = None
  147.         self.ehlo_resp = None
  148.         self.esmtp_features = {}
  149.         self.does_esmtp = 0
  150.     return (resp, reply)
  151.  
  152. class SSLFakeFile:
  153.     """A fake file like object that really wraps a SSLObject.
  154.  
  155.     It only supports what is needed in smtplib.
  156.     """
  157.     def __init__(self, sslobj):
  158.         self.sslobj = sslobj
  159.  
  160.     #pylint: disable=C0111
  161.     def readline(self):
  162.         string = ""
  163.         char = None
  164.         while char != "\n":
  165.             while True:
  166.                 try:          
  167.                     char = self.sslobj.recv(1)
  168.                     string += char
  169.                     break
  170.                 except gnutls.errors.OperationWouldBlock:
  171.                     time.sleep(0.1)
  172.         return string
  173.  
  174.     #pylint: disable=C0111
  175.     def close(self):
  176.         pass
  177.  
  178. smtplib.SMTP.starttls = starttls
  179.